package com.eddie.hashtab;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.ToString;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Scanner;

public class HashTabDemo {

    public static void main(String[] args) {
        HashTab hashTab = new HashTab(7);
        //简单的菜单
        String key;
        Scanner scanner = new Scanner(System.in);
        while (true) {
            System.out.println("add添加雇员");
            System.out.println("list显示");
            System.out.println("find根据Id查找Emp");
            System.out.println("exit退出");
            key = scanner.next();

            switch (key) {
                case "add" -> {
                    Emp emp = new Emp();
                    System.out.println("请输入雇员Id");
                    emp.id = scanner.nextInt();
                    System.out.println("请输入雇员姓名:");
                    emp.name = scanner.next();
                    hashTab.add(emp);
                }
                case "list" -> hashTab.list();
                case "find" -> {
                    System.out.println("请输入查找的Id");
                    hashTab.findEmpById(scanner.nextInt());
                }
                case "exit" -> {
                    return;
                }
            }
        }

    }

}

//创建hashTab 管理多条链表
class HashTab {
    private final EmpLinkedList[] empLinkedLists;
    private final int size;

    //构造器
    public HashTab(int size) {
        this.size = size;
        //初始化empLinkedLists
        empLinkedLists = new EmpLinkedList[size];
        for (int i = 0; i < size; i++) {
            empLinkedLists[i] = new EmpLinkedList();
        }
    }

    public void add(Emp emp) {
        int empLinkedListNo = hashFun(emp.id);
        //将emp添加到对应的链表中
        empLinkedLists[empLinkedListNo].add(emp);
    }

    //遍历所有的链表,遍历hash表
    public void list() {
        for (int i = 0; i < size; i++) {
            empLinkedLists[i].list(i + 1);
        }
    }

    //编写一个散列函数，使用一个取模法
    private int hashFun(int id) {
        return id % size;
    }

    /**
     * 根据输入的Id查找Id
     *
     * @param id
     */
    public void findEmpById(int id) {
        int i = hashFun(id);
        Emp empById = empLinkedLists[i].findEmpById(id);
        if (empById != null) {
            System.out.println("第" +(i + 1) + "条链表中找到了员工：" + empById);
        }else {
            System.out.println("查找失败");
        }
    }
}

//表示一个雇员
@AllArgsConstructor
@NoArgsConstructor
class Emp {
    public int id;
    public String name;
    public Emp next;

    public Emp(int id, String name) {
        super();
        this.id = id;
        this.name = name;
    }

    @Override
    public String toString() {
        return "Emp{" +
                "id=" + id +
                ", name='" + name + '\'' +
                "}";
    }
}

//创建EmpLinkedList，表示链表
class EmpLinkedList {
    //头指针指向第一个Emp，因此第一个链表是直接指向第一个雇员的
    private Emp head;

    /**
     * 添加雇员到链表
     * 说明：
     * 1、假设当添加雇员时，id是自增的，即id的分配总是从小到大
     *
     * @param emp 雇员
     */
    public void add(Emp emp) {
        if (head == null) {
            head = emp;
            return;
        }
        //如果不是第一个雇员，使用辅助执政定位到最后
        Emp curEmp = head;
        while (curEmp.next != null) {
            curEmp = curEmp.next;
        }
        //退出时直接将emp加入链表
        curEmp.next = emp;
    }

    //遍历链表信息
    public void list(int no) {
        //说明链表为空
        if (head == null) {
            System.out.println("第" + no + "链表为空");
            return;
        }
        System.out.println("第" + no + "链表信息：");
        Emp curEmp = head;
        while (curEmp != null) {
            System.out.println(curEmp);
            curEmp = curEmp.next;
        }
    }

    public boolean isEmpty() {
        return head == null;
    }

    public Emp findEmpById(int id) {
        if (isEmpty()) {
            System.out.println("链表为空");
            return null;
        }
        Emp curEmp = head;
        while (curEmp != null) {
            if (curEmp.id == id) {
                break;
            }
            //如果当前员工的下一位是空，表示当前员工为最后一位并且没有找到对应的员工，那么就将该员工设置空跳出循环返回
            if(curEmp.next==null) {
                curEmp = null;
                break;
            }
            curEmp = curEmp.next;
        }
        return curEmp;
    }
}
